package com.yswy.common.utils.page;

import com.baomidou.mybatisplus.core.metadata.IPage;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.io.Serializable;
import java.math.BigDecimal;
import java.util.List;
import java.util.stream.Collectors;

/**
 * 描述：分页工具类
 *
 * @author yanwei
 * @since  2023/7/3 9:44
 */
@Builder
@Data
@NoArgsConstructor
@AllArgsConstructor
public class PageUtils<T> implements Serializable {

    private static final long serialVersionUID = 1L;
    /**
     * 总记录数
     */
    private int totalCount;

    /**
     * 总页数
     */
    private int totalPage;
    /**
     * 当前记录起始索引
     */
    private int pageNum;

    /**
     * 每页显示记录数
     */
    private int pageSize;

    /**
     * 列表数据
     */
    private List<?> list;


    /**
     * 分页
     *
     * @param list       列表数据
     * @param totalCount 总记录数
     * @param pageSize   每页记录数
     * @param pageNum    当前页数
     */
    public PageUtils(List<T> list, int totalCount, int pageSize, int pageNum) {
        this.list = list;
        this.totalCount = totalCount;
        this.pageSize = pageSize;
        this.pageNum = pageNum;
        this.totalPage = (int) Math.ceil((double) totalCount / pageSize);
    }

    /**
     * 分页
     */
    public PageUtils(IPage<?> page) {
        this.list = page.getRecords();
        this.totalCount = (int) page.getTotal();
        this.pageSize = (int) page.getSize();
        this.pageNum = (int) page.getCurrent();
        this.totalPage = (int) page.getPages();
    }

    public static <T> PageUtils<T> result(IPage<?> page) {
        PageUtils<T> pageUtils = new PageUtils<>(page);
        pageUtils.setList(page.getRecords());
        return pageUtils;
    }

    /**
     * 描述：当特殊查询情况不能强转Page对象使用此方法
     *
     * @param pageList 分页数据
     * @param page     分页大小
     * @param pageSize 说明：
     * @return com.byd.common.core.page.PageResult<T>
     * @author xxx
     * @since  2022/9/28 16:38
     */
    public static <T> PageUtils<T> page(List<T> pageList, int page, int pageSize) {
        //手动分页
        PageUtils<T> pageResp = new PageUtils<>();
        int size = pageList.size();
        //分页
        List<T> collect = pageList.stream().
                skip((long) pageSize * (page - 1)).limit(pageSize).collect(Collectors.toList());
        pageResp.setList(collect);
        //设置分页信息
        pageResp.setTotalCount(size);
        pageResp.setPageNum(page);
        pageResp.setPageSize(pageSize);
        BigDecimal bigDecimal = new BigDecimal(size);
        BigDecimal divide = bigDecimal.divide(BigDecimal.valueOf(pageSize), 0, BigDecimal.ROUND_UP);
        //设置条数信息
        pageResp.setTotalPage(divide.intValue());
        return pageResp;
    }

}
